package com.tianji.aigc.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollStreamUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.RandomUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.tianji.aigc.config.SessionProperties;
import com.tianji.aigc.config.SystemPromptConfig;
import com.tianji.aigc.entity.ChatSession;
import com.tianji.aigc.enums.MessageTypeEnum;
import com.tianji.aigc.mapper.ChatSessionMapper;
import com.tianji.aigc.memory.MyAssistantMessage;
import com.tianji.aigc.service.ChatService;
import com.tianji.aigc.service.ChatSessionService;
import com.tianji.aigc.vo.ChatSessionVO;
import com.tianji.aigc.vo.MessageVO;
import com.tianji.aigc.vo.SessionVO;
import com.tianji.common.utils.CollUtils;
import com.tianji.common.utils.UserContext;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.ai.chat.client.ChatClient;
import org.springframework.ai.chat.memory.ChatMemory;
import org.springframework.ai.chat.messages.Message;
import org.springframework.ai.chat.messages.MessageType;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;

import java.time.LocalDateTime;
import java.time.temporal.ChronoUnit;
import java.util.List;
import java.util.Map;

@Slf4j
@Service
@RequiredArgsConstructor
public class ChatSessionServiceImpl extends ServiceImpl<ChatSessionMapper, ChatSession> implements ChatSessionService {


    private final SessionProperties sessionProperties;
    private final ChatMemory chatMemory;
    private final ChatClient titleClient;
    private final SystemPromptConfig systemPromptConfig;

    private static final String NEW_QUESTION = """
                对话内容：
                ------------
                USER:{} \\n
                ASSISTANT:{}
                ------------
                """;

    /**
     * 创建会话
     * @param num 示例数量
     * @return 会话信息
     */
    @Override
    public SessionVO createSession(Integer num) {
        SessionVO sessionVO = BeanUtil.toBean(sessionProperties, SessionVO.class);
        // 生成会话ID
        sessionVO.setSessionId(IdUtil.simpleUUID());
        // 随机获取热门话题
        sessionVO.setExamples(RandomUtil.randomEleList(sessionProperties.getExamples(), num));
        // 保存会话记录
        ChatSession chatSession = ChatSession.builder()
                .sessionId(sessionVO.getSessionId())
                .userId(UserContext.getUser())
                .build();
        super.save(chatSession);
        return sessionVO;
    }

    /**
     * 获取热门话题
     * @param num 示例数量
     * @return 会话信息
     */
    @Override
    public List<SessionVO.Example> hotExamples(Integer num) {
        return RandomUtil.randomEleList(sessionProperties.getExamples(), num);
    }

        /**
     * 根据会话ID查询会话记录
     * @param sessionId 会话ID
     * @return 会话记录列表，包含用户消息和助手消息
     */
    @Override
    public List<MessageVO> queryBySessionId(String sessionId) {
        //获取对话Id
        String conversationId = ChatService.getConversationId(sessionId);
        //获取会话记录
        List<Message> messages = chatMemory.get(conversationId);

        //过滤出用户消息和助手消息，并转换为MessageVO对象
        return messages.stream()
                .filter(message -> message.getMessageType() == MessageType.USER ||
                        message.getMessageType() == MessageType.ASSISTANT)
                .map(message ->{
                    //根据消息类型进行不同的VO构建处理
                    if (message instanceof MyAssistantMessage) {
                        return MessageVO.builder()
                                .type(MessageTypeEnum.valueOf(message.getMessageType().name()))
                                .content(message.getText())
                                .params(((MyAssistantMessage) message).getParams())
                                .build();
                    }
                    return MessageVO.builder()
                            .type(MessageTypeEnum.valueOf(message.getMessageType().name()))
                            .content(message.getText())
                            .build();
                })
                .toList();
    }

        /**
     * 更新会话标题
     * @param sessionId 会话ID
     * @param title 会话标题
     * @param userId 用户ID
     */
    @Async
    @Override
    public void update(String sessionId, String title, Long userId) {
        //查询会话记录
        ChatSession chatSession = super.lambdaQuery()
                .eq(ChatSession::getSessionId, sessionId)
                .eq(ChatSession::getUserId, userId)
                .one();
        if (chatSession == null) {
            return;
        }
        // 如果聊天会话的标题为空，并且新标题不为空，则更新标题
        if (StrUtil.isEmpty(chatSession.getTitle()) && StrUtil.isNotBlank(title)) {
            // 使用AI生成标题
            String content = titleClient.prompt()
                    .system(systemPromptConfig.getTitleSystemPrompt().get())
                    .user(title)
                    .call()
                    .content();
            log.info("用户{}会话{}的标题生成结果为{}", userId, sessionId, content);
            chatSession.setTitle(content);
        }
        // 更新会话的最后修改时间
        chatSession.setUpdateTime(LocalDateTime.now());
        super.updateById(chatSession);
    }


    /**
     * 查询历史会话
     * @return 历史会话列表
     */
    @Override
    public Map<String, List<ChatSessionVO>> queryHistory() {
        // 查询30条历史会话
        List<ChatSession> list = super.lambdaQuery()
                .eq(ChatSession::getUserId, UserContext.getUser())
                .orderByDesc(ChatSession::getUpdateTime)
                .isNotNull(ChatSession::getTitle)
                .last("limit 30")
                .list();
        // 若结果为空，则返回空
        if (CollUtils.isEmpty(list)) {
            return CollUtils.emptyMap();
        }
        // 将会话列表转换为ChatSessionVO对象
        List<ChatSessionVO> chatSessionVOList = CollStreamUtil.toList(list, chatSession ->
                ChatSessionVO
                        .builder()
                        .sessionId(chatSession.getSessionId())
                        .title(chatSession.getTitle())
                        .updateTime(chatSession.getUpdateTime())
                        .build()
        );

        final var TODAY = "当天";
        final var LAST_30_DAYS = "最近30天";
        final var LAST_YEAR = "最近1年";
        final var MORE_THAN_YEAR = "1年以上";

        // 当前时间
        var now = LocalDateTime.now().toLocalDate();

        return CollStreamUtil.groupByKey(chatSessionVOList, chatSessionVO -> {
            long abs = Math.abs(ChronoUnit.DAYS.between(chatSessionVO.getUpdateTime().toLocalDate(), now));
            if (abs <= 1) {
                return TODAY;
            } else if (abs <= 30) {
                return LAST_30_DAYS;
            } else if (abs <= 365) {
                return LAST_YEAR;
            } else {
                return MORE_THAN_YEAR;
            }
        });
    }

    /**
     * 删除历史会话
     * @param sessionId 会话ID
     */
    @Override
    public void deleteHistory(String sessionId) {
        // 删除数据库中会话记录
        super.remove(Wrappers.lambdaQuery(ChatSession.class)
                .eq(ChatSession::getSessionId, sessionId)
                .eq(ChatSession::getUserId, UserContext.getUser()));
        // 删除Redis中会话记录
        this.chatMemory.clear(ChatService.getConversationId(sessionId));
    }

        /**
     * 更新会话标题
     * @param sessionId 会话ID
     * @param title 新标题
     */
    @Override
    public void updateTitle(String sessionId, String title) {
        // 查询用户会话信息
        ChatSession chatSession = super.lambdaQuery()
                .eq(ChatSession::getSessionId, sessionId)
                .eq(ChatSession::getUserId, UserContext.getUser())
                .one();
        if (chatSession == null) {
            log.info("用户{}不存在的会话{}", UserContext.getUser(), sessionId);
            return;
        }

        // 更新会话标题（限制30个字符）
        if (StrUtil.isNotBlank(title)) {
            chatSession.setTitle(StrUtil.sub(title, 0, 30));
            super.updateById(chatSession);
            log.info("用户{}会话{}的标题已更新为{}", UserContext.getUser(), sessionId, title);
            return;
        }
        log.info("用户{}会话{}的新标题为空", UserContext.getUser(), sessionId);
    }


    /**
     * AI更新会话标题
     * @param sessionId 会话ID
     * @param aiStr AI返回的更新内容
     * @param question 用户的问题
     */
    @Override
    public void updateByAI(String sessionId, String aiStr, String question, Long userId) {
        update(sessionId, StrUtil.format(NEW_QUESTION, question, aiStr), userId);
    }
}
